python interview test example for data